
from aiClients.azureOpenAiClient import AzureOpenAiClient
from aiClients.baseAiClient import BaseAIClient, ModelType, ModelInfo
from aiClients.qwenAiClient import QwenAiClient


class ModelFactory:
    @staticmethod
    def create_model(model_type: str, model_info: ModelInfo) -> BaseAIClient:
        if model_type == ModelType.Azureopenai:
            return AzureOpenAiClient(model_info)
        elif model_type == ModelType.Qwen:
            return QwenAiClient(model_info)
        elif model_type == ModelType.Deepseek:
            return QwenAiClient(model_info)
        else:
            raise NotImplementedError
